split numpy array into train and test